package crawler.xiali;

import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.List;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.zql.entity.AgencyEntity;

import util.HttpConnectionPost;
import util.Location;
import util.MybatisTool;
import util.TianYanCha;

public class YiQiXiaLiCrawler {
	public List<AgencyEntity> getAgency() {
		List<AgencyEntity> list = new ArrayList<AgencyEntity>();
		try {
			String getPro = "http://www.tjfaw.com/index.php/JXSCX.shtml";
			Document proDoc = Jsoup.connect(getPro).get();
			Element proSelect = proDoc.getElementById("province");
			Elements proOptions = proSelect.getElementsByTag("option");
			for(int i=1;i<proOptions.size();i++) {
				String proName = proOptions.get(i).text();
				String proNum = proOptions.get(i).attr("value");
				
				//获取城市
				String getCity = "http://www.tjfaw.com/index.php/getCitys.shtml";
				String paramCiyt = "provinceID="+proNum;
				String cityText = HttpConnectionPost.getJson(getCity, paramCiyt);
				JSONArray cityArray = JSON.parseArray(cityText);
				for(int j=0;j<cityArray.size();j++) {
					JSONObject cityJSONObject = cityArray.getJSONObject(j);
					String cityName = cityJSONObject.getString("city");
					String cityNum = cityJSONObject.getString("cityID");
					
					//获取经销商
					String getDealer = "http://www.tjfaw.com/index.php/getfriend1.shtml";
					String paramDealer = "cityID="+cityNum;
					String dealerText = HttpConnectionPost.getJson(getDealer, paramDealer);
					
					JSONArray dealerArray = JSON.parseArray(dealerText);
					if(dealerArray == null) {
						continue;
					}
					for(int k=0;k<dealerArray.size();k++) {
						JSONObject dealer = dealerArray.getJSONObject(k);
						String address = dealer.getString("add");
						String name = dealer.getString("title");
						String sellTell = dealer.getString("tell");
						String[] lngAndLat = Location.getLocation(address);
						String lng = lngAndLat[0];
						String lat = lngAndLat[1];
						List<String> shareholder = TianYanCha.getShareholder(name);
						String sControllingShareholder = shareholder.get(0);
						String sOtherShareholders = shareholder.get(1);
						AgencyEntity agency = new AgencyEntity();
						agency.setdCloseDate(null);
						agency.setdOpeningDate(null);
						agency.setdUpdateTime(new Timestamp(System.currentTimeMillis()));
						agency.setnBrandID(-1);
						agency.setnDealerIDWeb(-1);
						agency.setsBrand("夏利");
						agency.setnManufacturerID(-1);
						agency.setsManufacturer("一汽夏利");
						agency.setnState(1);
						agency.setsAddress(address);
						agency.setsCity(cityName);
						agency.setsCustomerServiceCall(null);
						agency.setsDealerName(name);
						agency.setsDealerType(null);
						agency.setsProvince(proName);
						agency.setsSaleCall(sellTell);
						agency.setsLongitude(lng);
						agency.setsLatitude(lat);
						agency.setsControllingShareholder(sControllingShareholder);
						agency.setsOtherShareholders(sOtherShareholders);
						list.add(agency);
					}
				}
			}
		} catch(Exception e) {
			throw new RuntimeException("爬虫异常", e);
		}
		
		return list;
	}
	public static void main(String[] args) {
		YiQiXiaLiCrawler crawler = new YiQiXiaLiCrawler();
		System.out.println("爬虫开始...");
		List<AgencyEntity> agencys = crawler.getAgency();
		System.out.println("抓取完毕,正在存库");
		for(AgencyEntity agency : agencys) {
			MybatisTool.save(agency);
		}
		MybatisTool.close();
		System.out.println("请查看数据库");
	}

}
